package handler

import (
	"github.com/gin-gonic/gin"
	"path/filepath"
	"pic-text-adder/models"
)

func Add(c *gin.Context) {
	title := c.PostForm("title")
	content := c.PostForm("content")
	data := models.Article{

		Title:   title,
		Content: content,
	}
	if err := models.DB.Create(&data).Error; err != nil {
		c.JSON(200, gin.H{
			"code": 500,
			"msg":  "添加失败",
		})
		return
	}
	c.JSON(200, gin.H{
		"code": 200,
		"msg":  "添加成功",
	})
	return
}
func Upload(c *gin.Context) {
	file, _ := c.FormFile("file")
	ext := filepath.Ext(file.Filename)
	if ext != ".png" && ext != ".jpg" && ext != ".pneg" {
		c.JSON(200, gin.H{
			"code": 500,
			"msg":  "文件类型不正确",
		})
		return
	}
	dst := "./file" + file.Filename
	// 上传文件至指定的完整文件路径
	err := c.SaveUploadedFile(file, dst)
	if err != nil {
		c.JSON(200, gin.H{
			"code": 500,
			"msg":  "上传失败",
		})
		return
	}
	c.JSON(200, gin.H{
		"code": 200,
		"msg":  "上传成功",
	})
	return
}
func List(c *gin.Context) {
	var art []*models.Article
	if err := models.DB.Find(&art).Error; err != nil {
		c.JSON(200, gin.H{
			"code": 500,
			"msg":  "查询失败",
		})
		return
	}
	c.JSON(200, gin.H{
		"code": 200,
		"msg":  "查询成功",
		"data": art,
	})
	return
}
func Detail(c *gin.Context) {
	var art *models.Article
	id := c.Query("id")
	if err := models.DB.Where("id=?", id).First(&art).Error; err != nil {
		c.JSON(200, gin.H{
			"code": 500,
			"msg":  "查询失败",
		})
		return
	}
	c.JSON(200, gin.H{
		"code": 200,
		"msg":  "查询成功",
		"data": art,
	})
	return
}
